import os
import shutil

'''用于采集图片数据的重命名'''
# 创建新的文件夹
# os.makedirs('E:\mapping_img_change/data_change/rgb', exist_ok=True)
# os.makedirs('E:\mapping_img_change/data_change/depth', exist_ok=True)

# 读取aaa.txt文件内容
# 修改为寻找某一个路径下的名字为xx的文件夹
with open('E:\data\data_0905/room/associate.txt', 'r') as file:
    lines = file.readlines()

# 遍历文件内容
for index, line in enumerate(lines):
    # 解析每一行
    parts = line.strip().split()
    rgb_path = parts[1]
    depth_path = parts[3]
    # 添加前一部分文件夹的路径
    rgb_path = os.path.join('E:\data\data_0905key/room', rgb_path)
    depth_path = os.path.join('E:\data\data_0905key/room', depth_path)


    # 生成新的文件名
    new_filename = f"{index:06}.png"

    # 创建RGB文件夹（如果不存在）
    rgb_folder = 'E:\\change_data\\c_data_0905key\\rgb'
    if not os.path.exists(rgb_folder):
        os.makedirs(rgb_folder)

    # 创建Depth文件夹（如果不存在）
    depth_folder = 'E:\\change_data\\c_data_0905key\\depth'
    if not os.path.exists(depth_folder):
        os.makedirs(depth_folder)
    # 复制并重命名RGB文件
    new_rgb_path = os.path.join('E:\change_data\c_data_0905key/rgb', new_filename)
    shutil.copy(rgb_path, new_rgb_path)

    # 复制并重命名Depth文件
    new_depth_path = os.path.join('E:\change_data\c_data_0905key\depth', new_filename)
    shutil.copy(depth_path, new_depth_path)

    print(f"Renamed {rgb_path} to {new_rgb_path}")
    print(f"Renamed {depth_path} to {new_depth_path}")

print("Renaming completed!")
